package com.voltmao.webpro.utils.wechat.core;


import java.io.UnsupportedEncodingException;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.*;
import java.util.logging.Logger;

/**
 * Created by yangyincong on 15/12/7.
 */
public class Signature {
    private static Logger logger = Logger.getLogger(Signature.class.getName());

    public static String signatureSHA1(Map content, boolean onlyValue) throws NoSuchAlgorithmException, UnsupportedEncodingException {
        if (null == content || content.isEmpty()) {
            return "";
        }
        Map temp = new HashMap(content);
        temp.remove("signature");
        StringBuilder val = new StringBuilder("");
        if (onlyValue) {
            ArrayList list = new ArrayList(content.values());
            Collections.sort(list);
            for (Object o : list) {
                val.append(o.toString());
            }
        } else {
            //
            ArrayList list = new ArrayList(content.keySet());
            Collections.sort(list);
            for (Object o : list) {
                val.append(o.toString()).append("=").append(content.get(o).toString()).append("&");
            }
        }

        String result = val.toString();
        if (result.endsWith("&")) {
            result = result.substring(0, result.length() - 1);
        }
        logger.info("Pre sha1 encoding content : " + result);
        MessageDigest md = MessageDigest.getInstance("SHA1");
        md.update(result.getBytes("UTF-8"));
        byte[] bits = md.digest();
        return bytesToHex(bits);
    }

    public static String signatureMD5(Map<String, Object> params, String key) {
        params.remove("sign");
        if (null == key || key.length() < 1) {
            logger.warning("Pay info key not found..");
            return null;
        }

        ArrayList<String> list = new ArrayList<String>();
        for (Map.Entry<String, Object> entry : params.entrySet()) {
            if (entry.getValue() != "") {
                list.add(entry.getKey() + "=" + entry.getValue() + "&");
            }
        }
        int size = list.size();
        String[] arrayToSort = list.toArray(new String[size]);
        Arrays.sort(arrayToSort, String.CASE_INSENSITIVE_ORDER);
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < size; i++) {
            sb.append(arrayToSort[i]);
        }
        String result = sb.toString();
        logger.info("signatrueMD5:" + result);
        result += "key=" + key;
        MessageDigest md = null;
        try {
            md = MessageDigest.getInstance("MD5");
            md.update(result.getBytes("UTF-8"));

        } catch (NoSuchAlgorithmException e) {
            e.printStackTrace();
        } catch (UnsupportedEncodingException e) {
            e.printStackTrace();
        }

        result = bytesToHex(md.digest());
        logger.info("{" + result + "} md5 sign result {" + result + "}");
        return result;
    }

    public static String bytesToHex(byte[] b) {
        char hexDigit[] = {'0', '1', '2', '3', '4', '5', '6', '7',
                '8', '9', 'A', 'B', 'C', 'D', 'E', 'F'};
        StringBuffer buf = new StringBuffer();
        for (int j = 0; j < b.length; j++) {
            buf.append(hexDigit[(b[j] >> 4) & 0x0f]);
            buf.append(hexDigit[b[j] & 0x0f]);
        }
        return buf.toString();
    }

    public static String signSHA1(Map<String, String> map) throws NoSuchAlgorithmException, UnsupportedEncodingException {

        String[] ArrTmp = {map.get("token"), map.get("timestamp"), map.get("nonce")};
        Arrays.sort(ArrTmp);
        StringBuffer sb = new StringBuffer();
        for (int i = 0; i < ArrTmp.length; i++) {
            sb.append(ArrTmp[i]);
        }


        return new SHA1().getDigestOfString(sb.toString().getBytes()).toLowerCase();
    }

    /**
     * 获取32位随机字符串
     *
     * @return
     */
    public static String randomString() {
        return UUID.randomUUID().toString().replace("-", "");
    }


    /**
     * 测试
     *
     * @param args
     */
    public static void main(String[] args) throws UnsupportedEncodingException, NoSuchAlgorithmException {

//        String signature="4fQHaou6TGPcUuo79KKYonTzPAmPQntoJSCd48GRcbF";
//
//        Map<String,String> map =  new HashMap<String,String>();
//        map.put("timestamp","1365760417");
//        map.put("nonce","1365691777");
//        map.put("token","comwechat");
//
//
//
//
//        String pwd =Signature.signatureMD5(map);
//        System.out.println(pwd);
//        if (signature.equals(pwd)) {
//            System.out.println("token 验证成功~!");
//        }else {
//            System.out.println("token 验证失败~!");
//        }


    }
}

class SHA1 {
    private final int[] abcde = {0x67452301, 0xefcdab89, 0x98badcfe,
            0x10325476, 0xc3d2e1f0};
    // ժҪ���ݴ洢����
    private int[] digestInt = new int[5];
    // ��������е���ʱ���ݴ洢����
    private int[] tmpData = new int[80];

    // ����sha-1ժҪ
    private int process_input_bytes(byte[] bytedata) {
        // ���Ի�����
        System.arraycopy(abcde, 0, digestInt, 0, abcde.length);
        // ��ʽ�������ֽ����飬��10����������
        byte[] newbyte = byteArrayFormatData(bytedata);
        // ��ȡ����ժҪ��������ݵ�Ԫ����
        int MCount = newbyte.length / 64;
        // ѭ����ÿ�����ݵ�Ԫ����ժҪ����
        for (int pos = 0; pos < MCount; pos++) {
            // ��ÿ����Ԫ������ת����16���������ݣ������浽tmpData��ǰ16������Ԫ����
            for (int j = 0; j < 16; j++) {
                tmpData[j] = byteArrayToInt(newbyte, (pos * 64) + (j * 4));
            }
            // ժҪ���㺯��
            encrypt();
        }
        return 20;
    }

    // ��ʽ�������ֽ������ʽ
    private byte[] byteArrayFormatData(byte[] bytedata) {
        // ��0����
        int zeros = 0;
        // ��λ����λ��
        int size = 0;
        // ԭʼ���ݳ���
        int n = bytedata.length;
        // ģ64���ʣ��λ��
        int m = n % 64;
        // �������0�ĸ����Լ����10����ܳ���
        if (m < 56) {
            zeros = 55 - m;
            size = n - m + 64;
        } else if (m == 56) {
            zeros = 63;
            size = n + 8 + 64;
        } else {
            zeros = 63 - m + 56;
            size = (n + 64) - m + 64;
        }
        // ��λ�����ɵ�����������
        byte[] newbyte = new byte[size];
        // ���������ǰ�沿��
        System.arraycopy(bytedata, 0, newbyte, 0, n);
        // �������Append����Ԫ�ص�λ��
        int l = n;
        // ��1����
        newbyte[l++] = (byte) 0x80;
        // ��0����
        for (int i = 0; i < zeros; i++) {
            newbyte[l++] = (byte) 0x00;
        }
        // �������ݳ��ȣ������ݳ���λ��8�ֽڣ�������
        long N = (long) n * 8;
        byte h8 = (byte) (N & 0xFF);
        byte h7 = (byte) ((N >> 8) & 0xFF);
        byte h6 = (byte) ((N >> 16) & 0xFF);
        byte h5 = (byte) ((N >> 24) & 0xFF);
        byte h4 = (byte) ((N >> 32) & 0xFF);
        byte h3 = (byte) ((N >> 40) & 0xFF);
        byte h2 = (byte) ((N >> 48) & 0xFF);
        byte h1 = (byte) (N >> 56);
        newbyte[l++] = h1;
        newbyte[l++] = h2;
        newbyte[l++] = h3;
        newbyte[l++] = h4;
        newbyte[l++] = h5;
        newbyte[l++] = h6;
        newbyte[l++] = h7;
        newbyte[l++] = h8;
        return newbyte;
    }

    private int f1(int x, int y, int z) {
        return (x & y) | (~x & z);
    }

    private int f2(int x, int y, int z) {
        return x ^ y ^ z;
    }

    private int f3(int x, int y, int z) {
        return (x & y) | (x & z) | (y & z);
    }

    private int f4(int x, int y) {
        return (x << y) | x >>> (32 - y);
    }

    // ��ԪժҪ���㺯��
    private void encrypt() {
        for (int i = 16; i <= 79; i++) {
            tmpData[i] = f4(tmpData[i - 3] ^ tmpData[i - 8] ^ tmpData[i - 14]
                    ^ tmpData[i - 16], 1);
        }
        int[] tmpabcde = new int[5];
        for (int i1 = 0; i1 < tmpabcde.length; i1++) {
            tmpabcde[i1] = digestInt[i1];
        }
        for (int j = 0; j <= 19; j++) {
            int tmp = f4(tmpabcde[0], 5)
                    + f1(tmpabcde[1], tmpabcde[2], tmpabcde[3]) + tmpabcde[4]
                    + tmpData[j] + 0x5a827999;
            tmpabcde[4] = tmpabcde[3];
            tmpabcde[3] = tmpabcde[2];
            tmpabcde[2] = f4(tmpabcde[1], 30);
            tmpabcde[1] = tmpabcde[0];
            tmpabcde[0] = tmp;
        }
        for (int k = 20; k <= 39; k++) {
            int tmp = f4(tmpabcde[0], 5)
                    + f2(tmpabcde[1], tmpabcde[2], tmpabcde[3]) + tmpabcde[4]
                    + tmpData[k] + 0x6ed9eba1;
            tmpabcde[4] = tmpabcde[3];
            tmpabcde[3] = tmpabcde[2];
            tmpabcde[2] = f4(tmpabcde[1], 30);
            tmpabcde[1] = tmpabcde[0];
            tmpabcde[0] = tmp;
        }
        for (int l = 40; l <= 59; l++) {
            int tmp = f4(tmpabcde[0], 5)
                    + f3(tmpabcde[1], tmpabcde[2], tmpabcde[3]) + tmpabcde[4]
                    + tmpData[l] + 0x8f1bbcdc;
            tmpabcde[4] = tmpabcde[3];
            tmpabcde[3] = tmpabcde[2];
            tmpabcde[2] = f4(tmpabcde[1], 30);
            tmpabcde[1] = tmpabcde[0];
            tmpabcde[0] = tmp;
        }
        for (int m = 60; m <= 79; m++) {
            int tmp = f4(tmpabcde[0], 5)
                    + f2(tmpabcde[1], tmpabcde[2], tmpabcde[3]) + tmpabcde[4]
                    + tmpData[m] + 0xca62c1d6;
            tmpabcde[4] = tmpabcde[3];
            tmpabcde[3] = tmpabcde[2];
            tmpabcde[2] = f4(tmpabcde[1], 30);
            tmpabcde[1] = tmpabcde[0];
            tmpabcde[0] = tmp;
        }
        for (int i2 = 0; i2 < tmpabcde.length; i2++) {
            digestInt[i2] = digestInt[i2] + tmpabcde[i2];
        }
        for (int n = 0; n < tmpData.length; n++) {
            tmpData[n] = 0;
        }
    }

    // 4�ֽ�����ת��Ϊ����
    private int byteArrayToInt(byte[] bytedata, int i) {
        return ((bytedata[i] & 0xff) << 24) | ((bytedata[i + 1] & 0xff) << 16)
                | ((bytedata[i + 2] & 0xff) << 8) | (bytedata[i + 3] & 0xff);
    }

    // ����ת��Ϊ4�ֽ�����
    private void intToByteArray(int intValue, byte[] byteData, int i) {
        byteData[i] = (byte) (intValue >>> 24);
        byteData[i + 1] = (byte) (intValue >>> 16);
        byteData[i + 2] = (byte) (intValue >>> 8);
        byteData[i + 3] = (byte) intValue;
    }

    // ���ֽ�ת��Ϊʮ�������ַ���
    private static String byteToHexString(byte ib) {
        char[] Digit = {'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'A',
                'B', 'C', 'D', 'E', 'F'};
        char[] ob = new char[2];
        ob[0] = Digit[(ib >>> 4) & 0X0F];
        ob[1] = Digit[ib & 0X0F];
        String s = new String(ob);
        return s;
    }

    // ���ֽ�����ת��Ϊʮ�������ַ���
    private static String byteArrayToHexString(byte[] bytearray) {
        String strDigest = "";
        for (int i = 0; i < bytearray.length; i++) {
            strDigest += byteToHexString(bytearray[i]);
        }
        return strDigest;
    }

    // ����sha-1ժҪ��������Ӧ���ֽ�����
    public byte[] getDigestOfBytes(byte[] byteData) {
        process_input_bytes(byteData);
        byte[] digest = new byte[20];
        for (int i = 0; i < digestInt.length; i++) {
            intToByteArray(digestInt[i], digest, i * 4);
        }
        return digest;
    }

    // ����sha-1ժҪ��������Ӧ��ʮ�������ַ���
    public String getDigestOfString(byte[] byteData) {
        return byteArrayToHexString(getDigestOfBytes(byteData));
    }

    public static void main(String[] args) {
        String data = "123456";
        System.out.println(data);
        String digest = new SHA1().getDigestOfString(data.getBytes());
        System.out.println(digest);

        // System.out.println( ToMD5.convertSHA1(data).toUpperCase());
    }
}